from pydantic import BaseModel
import time
from typing import Optional, List
import random


class Parameter(BaseModel):
    node_ids: List[float]


class Result(BaseModel):
    displacement: List[float]


def random_num(num=200):
    return [random.random() for _ in range(num)]


def main(param: Parameter) -> Result:
    time.sleep(20)
    return Result(displacement=random_num(20))
